import os
#定义获取目录大小的函数
def getdirsize(dir):
	#获取目标目录中的所有文件信息
	dlist=os.listdir(dir)
	#遍历所有文件，并计算文件大小
	dirsize=0
	for f in dlist:
		#为遍历的文件添加目录路径
		file=os.path.join(dir,f)
		#判断是否是文件
		if os.path.isfile(file):
			a=os.path.getsize(file)
			dirsize+=a
		#判断是否是目录
		if os.path.isdir(file):
			b=getdirsize(file)#递归调用自己，实现子目录大小的计算
			dirsize+=b
	return dirsize
	
a=input("please enter dir:")
c=getdirsize(a)
print("the size is:{}".format(c))









